司开星的博客

Python 的列表推导式、字典推导式

列表推导式

列表解析是用一行代码实现根据一个列表生成另一个列表的写法。

基本写法

要实现从一个列表中获取满足条件的某些元素组成新的列表:

[x for x in [1,2,3,4] if x % 2 == 0]

这行代码输出是 [2, 4]。如果没有 if 的话就是直接复制整个列表。

其他用法

修改新列表的元素:

[x/2 for x in [1,2,3,4]]

将单个元素的列表重复此元素,扩展为指定长度的列表:

[x for i in range(100)]

提取嵌套列表中元素的第一项:

[x[0] for x in [[1,2], [3,4], [5,6]]]

嵌套循环的写法:

[n for row in [[1,2], [3,4], [5,6]] for n in row]

生成嵌套列表:

[(x,x+10) for x in [1,2]

多个参数:
[(x,y) for x,y in {1:2, 4:5}.items() if x % 2 == 0]

与条件表达式结合:
[x if x % 3 == 0 else 1 for x in range(10)]

[(x,y) if x % 2 == 0 else (y, 3) for x,y in {1:2, 4:5}.items() ]

字典推导式

与列表推导式类似,字典也有相应的推导式写法。

基本写法

{x:y for y,x in {'1':'2', '3':'4'}.items() if '1' in y}

参考资料:

《轻松学会Python列表解析式》
《字典推导式》